package org.basis.algorithm;

import org.basis.algorithm.graph.common.Node;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Stack;

/**
 * 深度优先遍历
 *
 * @author Mr_wenpan@163.com 2021/12/23 16:05
 */
public class Dfs {

    public static void main(String[] args) {

    }

    public static List<Node> dfs(Node node) {
        if (node == null) {
            return null;
        }
        Stack<Node> stack = new Stack<>();
        HashSet<Node> set = new HashSet<>();
        List<Node> res = new ArrayList<>();
        stack.push(node);
        set.add(node);
        res.add(node);

        while (!stack.isEmpty()) {
            Node pop = stack.pop();
            for (Node next : pop.nexts) {
                // 逮住一个没走过的节点一直走到死
                if (!set.contains(next)) {
                    set.add(next);
                    // 先把当前节点压回去，因为他的其他next节点可能还没看过
                    stack.push(pop);
                    stack.push(next);
                    res.add(next);
                    break;
                }
            }
        }

        return res;
    }

}
